# guestfish bash completion script -*- shell-script -*-
# Copyright (C) 2010-2023 Red Hat Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

# List all local libvirt domains.
_guestfs_virsh_list ()
{
    local flag_ro=$1 flags

    if [ "$flag_ro" -eq 1 ]; then
        flags="--all"
    else
        flags="--inactive"
    fi
    virsh list $flags | head -n -1 | tail -n +3 | awk '{print $2}'
}

_guestfish ()
{
    local cur prev words cword split
    local shortopts longopts flag_ro=0 c=1 word cmds doms

    _init_completion -s || return

    # See if user has specified certain options anywhere on the
    # command line before the current word.
    while [ $c -lt $COMP_CWORD ]; do
        word="${COMP_WORDS[c]}"
        case "$word" in
            -r|--ro) flag_ro=1 ;;
        esac
        c=$((++c))
    done

    case "$prev" in
        -a|--add)
            COMPREPLY=( $(compgen "$cur") )
            return ;;
        -d|--domain)
            doms=$(_guestfs_virsh_list "$flag_ro")
            COMPREPLY=( $(compgen -W "$doms" -- "$cur") )
            return ;;
    esac

    case "$cur" in
        --*)
            # --options
            longopts="$(guestfish --long-options)"
            COMPREPLY=( $(compgen -W "$longopts" -- "$cur") )
            return ;;
        -*)
            # -o and --options
            shortopts="$(guestfish --short-options)"
            longopts="$(guestfish --long-options)"
            COMPREPLY=( $(compgen -W "$shortopts $longopts" -- "$cur") )
            return ;;
        *)
            # Might be a guestfish command.
            cmds=$(guestfish -h| head -n -1 | tail -n +2 | awk '{print $1}')
            COMPREPLY=( $(compgen -W "$cmds" -- "$cur") )
            return ;;
    esac
}

complete -o default -F _guestfish guestfish

# virt-copy-in etc are literally the same as guestfish
complete -o default -F _guestfish virt-copy-in
complete -o default -F _guestfish virt-copy-out
complete -o default -F _guestfish virt-tar-in
complete -o default -F _guestfish virt-tar-out
